defining function in python